//给定一个整数数组和一个整数 k，你需要在数组里找到 不同的 k-diff 数对，并返回不同的 k-diff 数对 的数目。 
//
// 这里将 k-diff 数对定义为一个整数对 (nums[i], nums[j])，并满足下述全部条件： 
//
// 
// 0 <= i < j < nums.length 
// |nums[i] - nums[j]| == k 
// 
//
// 注意，|val| 表示 val 的绝对值。 
//
// 
//
// 示例 1： 
//
// 
//输入：nums = [3, 1, 4, 1, 5], k = 2
//输出：2
//解释：数组中有两个 2-diff 数对, (1, 3) 和 (3, 5)。
//尽管数组中有两个1，但我们只应返回不同的数对的数量。
// 
//
// 示例 2： 
//
// 
//输入：nums = [1, 2, 3, 4, 5], k = 1
//输出：4
//解释：数组中有四个 1-diff 数对, (1, 2), (2, 3), (3, 4) 和 (4, 5)。
// 
//
// 示例 3： 
//
// 
//输入：nums = [1, 3, 1, 5, 4], k = 0
//输出：1
//解释：数组中只有一个 0-diff 数对，(1, 1)。
// 
//
// 示例 4： 
//
// 
//输入：nums = [1,2,4,4,3,3,0,9,2,3], k = 3
//输出：2
// 
//
// 示例 5： 
//
// 
//输入：nums = [-1,-2,-3], k = 1
//输出：2
// 
//
// 
//
// 提示： 
//
// 
// 1 <= nums.length <= 104 
// -107 <= nums[i] <= 107 
// 0 <= k <= 107 
// 
// Related Topics 数组 哈希表 双指针 二分查找 排序 
// 👍 153 👎 0


package leetcode.editor.cn.lc532;

import java.time.Instant;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

public class KDiffPairsInAnArray{
    public static void main(String[] args){
        Solution solution = new Solution();
        int pairs = solution.findPairs(new int[]{3,1,4,1,5}, 2);
        System.out.println("pairs = " + pairs);
    }
}
//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
    public int findPairs2(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int num : nums) {
            map.compute(num, (key, count) -> count == null ? 1 : count + 1);
        }

        if (k == 0) {
            return (int) map.values().stream().filter(v -> v > 1).count();
        }

        int total = 0;

        for (Integer num : map.keySet()) {
            int big = num + k;
            if (map.containsKey(big)) {
                total++;
            }
        }

        return total;

    }

    public int findPairs(int[] nums, int k) {

        Arrays.sort(nums);

        int total = 0;

        int cur = Integer.MIN_VALUE;
        for (int i = 0; i < nums.length; i++) {
            int num = nums[i];
            if (num == cur) {
                continue;
            }
            cur = num;
            int i1 = Arrays.binarySearch(nums, i + 1, nums.length, num + k);
            if (i1 >= 0) {
                total++;
            }
        }

        return total;

    }
}
//leetcode submit region end(Prohibit modification and deletion)

